package club.xiaojiawei.back;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * @author 肖嘉威
 * @version 1.0
 * @date 5/21/22 5:33 PM
 * @question 90. 子集 II
 * @description 给你一个整数数组 nums ，其中可能包含重复元素，请你返回该数组所有可能的子集（幂集）。
 * 解集 不能 包含重复的子集。返回的解集中，子集可以按 任意顺序 排列。
 */
public class SubsetsWithDup90 {

    public static void main(String[] args) {
        SubsetsWithDup90 test = new SubsetsWithDup90();
        System.out.println(test.subsetsWithDup(new int[]{1,2,2}));
    }

    /**
     * 递归回溯
     * @param nums
     * @return
     */
    public List<List<Integer>> subsetsWithDup(int[] nums) {
        result.add(new ArrayList<>());
        Arrays.sort(nums);
        recursion(nums, 0);
        return result;
    }

    List<List<Integer>> result = new ArrayList<>();

    List<Integer> temp = new ArrayList<>();

    public void recursion(int[] nums, int startIndex){
        if (startIndex == nums.length){
            return;
        }
        for (int i = startIndex; i < nums.length ; i++){
            if ((i != startIndex && nums[i] == nums[i - 1])){
                continue;
            }
            temp.add(nums[i]);
            result.add(new ArrayList<>(temp));
            recursion(nums, i + 1);
            temp.remove(temp.size() - 1);
        }
    }

    List<Integer> t = new ArrayList<Integer>();
    List<List<Integer>> ans = new ArrayList<List<Integer>>();

    /**
     * 官方-迭代
     * @param nums
     * @return
     */
    public List<List<Integer>> subsetsWithDup2(int[] nums) {
        Arrays.sort(nums);
        int n = nums.length;
        for (int mask = 0; mask < (1 << n); ++mask) {
            t.clear();
            boolean flag = true;
            for (int i = 0; i < n; ++i) {
                if ((mask & (1 << i)) != 0) {
//                   若发现没有选择上一个数，且当前数字与上一个数相同，则可以跳过当前生成的子集。
                    if (i > 0 && (mask >> (i - 1) & 1) == 0 && nums[i] == nums[i - 1]) {
                        flag = false;
                        break;
                    }
                    t.add(nums[i]);
                }
            }
            if (flag) {
                ans.add(new ArrayList<Integer>(t));
            }
        }
        return ans;
    }

    /**
     * 官方-递归回溯
     * @param nums
     * @return
     */
    public List<List<Integer>> subsetsWithDup3(int[] nums) {
        Arrays.sort(nums);
        dfs(false, 0, nums);
        return ans;
    }

    public void dfs(boolean choosePre, int cur, int[] nums) {
        if (cur == nums.length) {
            ans.add(new ArrayList<Integer>(t));
            return;
        }
        dfs(false, cur + 1, nums);
        if (!choosePre && cur > 0 && nums[cur - 1] == nums[cur]) {
            return;
        }
        t.add(nums[cur]);
        dfs(true, cur + 1, nums);
        t.remove(t.size() - 1);
    }

}
